# -*- encoding: utf-8 -*-
"""

PyCharm scipy库的使用
2024年10月22日21时23分
by wenyang

文件主要内容:


"""
from scipy import fftpack as fp
import numpy as np
import matplotlib.pyplot as plt

def main():
    t = np.arange(0, 1, 0.001)
    sig = np.sin(2 * np.pi * t + np.pi)
    sig += np.random.randn(sig.size) * 0.2
    sig_fft = fp.fft(sig)
    freqs = fp.fftfreq(sig.size, d = 0.001)

    plt.figure(figsize = (16, 4))
    plt.subplot(121)
    plt.plot(t, sig)
    plt.subplot(122)
    plt.plot(freqs, np.abs(sig_fft))
    plt.show()




if __name__ == "__main__":
    main()
